from locust import HttpUser, between
from test_cases.test_login import LoginTest
from test_cases.test_data_api import DataAPITest
from test_cases.test_submission import SubmitTest
from configs import settings

class APIUser(LoginTest, DataAPITest, SubmitTest):
    wait_time = between(1, 3)

if __name__ == "__main__":
    import os
    os.system(f"locust -f locustfile.py --headless --users {settings.USERS} --spawn-rate {settings.SPAWN_RATE} --run-time {settings.RUN_TIME}")
